package com.zboin.interview150;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/** Normal
 * 给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。
 * 示例 1：
 * 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
 * 输出：[1,2,3,6,9,8,7,4,5]
 * 示例 2：
 * 输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
 * 输出：[1,2,3,4,8,12,11,10,9,5,6,7]
 * 提示：
 * m == matrix.length
 * n == matrix[i].length
 * 1 <= m, n <= 10
 * -100 <= matrix[i][j] <= 100
 */
public class Course54 {

    public static void main(String[] args) {
        Course54 course54 = new Course54();

//        int[][] martix = {{1,2,3},{4,5,6},{7,8,9}};
        int[][] martix = {{1,2,3,4},{5,6,7,8},{9,10,11,12}};

        List<Integer> list = course54.spiralOrder(martix);
        System.out.println(Arrays.toString(list.toArray(new Integer[0])));
    }

    public List<Integer> spiralOrder(int[][] matrix) {
        int row = matrix.length, col = matrix[0].length;
        List<Integer> list = new ArrayList<>();

        // flag
        // 0: 表示x轴向右移动 -->
        // 1: 表示y轴向下移动 --> 1
        // 2: 表示x轴向左移动 --> 2
        // 3: 表示y轴向上移动 --> 3
        int flag = 0;
        int c = 0, n = row * col, x = 0, y = 0;
        int xStart = 0, yStart = 0, xEnd = row, yEnd = col;
        while (c < n) {
            if (flag == 0) {
                for (; y < yEnd; y++) {
                    list.add(matrix[x][y]);
                    c++;
                }
                xStart++; x++; y--; flag++;
            } else if (flag == 1) {
                for (; x < xEnd; x++) {
                    list.add(matrix[x][y]);
                    c++;
                }
                yEnd--; y--; x--; flag++;
            } else if (flag == 2) {
                for (; y >= yStart; y--) {
                    list.add(matrix[x][y]);
                    c++;
                }
                xEnd--; x--; y++; flag++;
            } else if (flag == 3) {
                for (; x >= xStart; x--) {
                    list.add(matrix[x][y]);
                    c++;
                }
                yStart++; y++; x++; flag++;
            } else {
                flag = 0;
            }
        }

        return list;
    }
}
